from betting.main.models import UserProfile, BetModel, BankModel, UserBetsModel
from django.contrib import admin
from django.contrib.auth.models import User


class ProfileAdmin(admin.ModelAdmin):
    list_display = ('username', 'email')
    
class BetAdmin(admin.ModelAdmin):
    list_display = ('title', 'description', 'price','active')

class BankAdmin(admin.ModelAdmin):
    list_display = ('user','bankroll','current_bets')
    
class UserBetsAdmin(admin.ModelAdmin):
    list_display= ('user','bet')

admin.site.register(UserProfile, ProfileAdmin)
admin.site.register(BetModel, BetAdmin)
admin.site.register(BankModel, BankAdmin)
admin.site.register(UserBetsModel, UserBetsAdmin)